tensor([[0.3300, 0.3300, 0.3300, 0.3300, 0.3300],
        [0.6700, 0.6700, 0.6700, 0.6700, 0.6700]], grad_fn=<RoundBackward1>)